"""
@Project ：TpShopVerifyCodeOCR 
@File    ：GetMyWebCookieTest.py
@IDE     ：PyCharm 
@Author  ：张成龙
@Date    ：2024/10/10 15:55 
@explain ：
"""

import os
import time

from dotenv import load_dotenv, set_key
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.edge.options import Options
from selenium.webdriver.edge.service import Service
from webdriver_manager.microsoft import EdgeChromiumDriverManager

# 创建 EdgeOptions 对象
# 创建 WebDriver 实例
options = Options()
options.add_argument('--headless')
driver = webdriver.Edge(service=Service(EdgeChromiumDriverManager().install()), options=options)

driver.implicitly_wait(10)
driver.get("http://192.168.56.128:8090/console/login")


driver.find_element(By.XPATH, "//*[@id='input_0']").send_keys("admin")
driver.find_element(By.ID, "passwordInput").send_keys("admin")

driver.find_element(By.XPATH, "//*[@id='app']/div/div[1]/button/span").click()

time.sleep(2)
cookies = driver.get_cookies()
print(cookies)
# 获取 session 值
session_value = cookies[0]["value"]

print("SESSION=", session_value)


# 检查 .env 文件是否存在，如果不存在则创建
env_file = '.env'

if not os.path.exists(env_file):
    with open(env_file, 'w') as f:
        pass  # 创建空的 .env 文件

# 加载现有的 .env 文件
load_dotenv()

# 将 SESSION 写入 .env 文件（更新或添加）
# quote_mode='never' SESSION 值不添加后面的 ''
set_key(env_file, 'SESSION', session_value, quote_mode='never')

